package com.at.bigdata.spark.core.rdd.operator.transform

import org.apache.spark.{SparkConf, SparkContext}

/**
 *
 * @author cdhuangchao3
 * @date 2023/3/19 9:52 PM
 */
object Spark08_RDD_sample_Test {

  def main(args: Array[String]): Unit = {
    // 环境准备
    val sparkConf = new SparkConf().setMaster("local[*]").setAppName("Operator")
    val sc = new SparkContext(sparkConf)

    // TODO 算子 - sample -- 场景：解决分区倾斜
    val rdd = sc.makeRDD(List(1,2,3,4,5,6,7,8,9,10))

    // param1： 抽取后数据是否放回， true-放回，false-丢弃
    // param2： 数据源中，
    //      抽取不放回场合：每条数据被抽取的概率，基准值的概念
    //      抽取放回长虹：  每条数据被抽取的可能次数
    // param3： 抽取数据时，随机算法的种子
    //          如果不传递第三个参数，使用当前系统时间
    println(rdd.sample(
      false,
      0.4,
//      1
    ).collect().mkString(","))

    println(rdd.sample(
      true,
      2
    ).collect().mkString(","))

    sc.stop()
  }

}
